from app.routes import ClasRoutes
class ClasView:
    def __init__(self):
        self.function_list=["查看班级信息","添加班级信息","修改班级信息","删除班级信息"]
    def display_menu(self):
        print("\n============== 学生成绩管理系统-班级管理 ==============")
        for idx, func in enumerate(self.function_list):
            print(f"{idx+1}.{func}")
        print("5.返回")
        print("请根据上面的选项输入数字并按回车继续...")

    def get_clases(self):
        '''
        用户点击查看班级信息
        :return:
        '''
        keyword = input("请输入班级关键字，回车搜索全部：")
    #     查询所有的班级信息
        resp = ClasRoutes.get_by_keyword(keyword)
        # 打印查询到的信息
        print(resp['message'])
        for data in resp["data"]:
            print(data)

    def add_class(self):
        '''
        选择功能添加班级管理视图
        :return:
        '''
        clas_name = input("请输入班级名：")
        resp = ClasRoutes.add_clas(clas_name)
        print(resp['message'])

    def update_class(self):
        self.get_clases()
        clas_id = input("请输入你要修改的班级id：")
        new_name = input("请输入新的班级名:")
        resp = ClasRoutes.update_clas(clas_id, new_name)
        print(resp['message'])

    def delete_clas(self):
        self.get_clases()
        clas_id = input("请输入你要删除的班级id：")
        resp = ClasRoutes.delete_clas(clas_id)
        print(resp['message'])

    def run(self):
        self.display_menu()
        choice = input("请输入操作符:")
        if choice == "1":
            self.get_clases()
            self.run()
        elif choice == "2":
            self.add_class()
            self.run()
        elif choice == "3":
            self.update_class()
            self.run()
        elif choice == "4":
            self.delete_clas()
            self.run()
        elif choice == "5":
            pass
        else:
            print("请输入正确的操作符")